# 使用文件和目录操作，定义一个统计指定目录大小的函数（注意目录中还有子 目录）。
import os

# 调用os模块统计目录大小
def dirSize(dir):
    sum = 0  # 初始化sum存储大小
    # 判断是文件还是目录，文件直接获取大小，目录则进入下级循环
    if os.path.isfile(dir):
        sum += os.path.getsize(dir)
    if os.path.isdir(dir):
        # 列出子目录for循环进行遍历
        listDir = os.listdir(dir)
        for i in listDir:
            # 做目录字符串拼接方便if进行判断
            file = os.path.join(dir,i)
            #print(file)
            if os.path.isdir(file):
                sum += dirSize(file)
            if os.path.isfile(file):
                sum += os.path.getsize(file)
    return sum

dir = input("请输入需要统计大小的目录： ")
sum = dirSize(dir)
print("目录的大小为： %d MB\t " %(sum/1024/1024))
